import os
import PyInstaller.__main__
import shutil

def build():
    # 删除之前的构建文件
    if os.path.exists('build'):
        shutil.rmtree('build')
    if os.path.exists('dist'):
        shutil.rmtree('dist')
    
    # PyInstaller 参数
    args = [
        'main.py',  # 主程序入口
        '--name=大唐无双',  # 生成的exe名称
        '--windowed',  # 使用 GUI 模式
        '--icon=resources/icon.ico',  # 程序图标（需要你提供一个图标文件）
        '--add-data=config;config',  # 添加配置文件目录
        '--noconfirm',  # 不询问确认
        '--clean',  # 清理临时文件
        '--noconsole',  # 不显示控制台
    ]
    
    # 运行 PyInstaller
    PyInstaller.__main__.run(args)
    
    # 复制额外的文件到打包目录
    dist_dir = os.path.join('dist', '大唐无双')
    
    # 复制配置文件
    if not os.path.exists(os.path.join(dist_dir, 'config')):
        shutil.copytree('config', os.path.join(dist_dir, 'config'))

if __name__ == '__main__':
    build() 